#创建一个浏览器对象
import datetime
import os.path
import sys

from selenium import webdriver
from selenium.webdriver.edge.options import Options
from selenium.webdriver.edge.service import Service
from webdriver_manager.microsoft import EdgeChromiumDriverManager


class Driver:
    def __init__(self):
        # 1. 自动下载并配置 Edge WebDriver
        self.driver_path = EdgeChromiumDriverManager().install()

        # 2. 配置 Edge 浏览器选项
        self.edge_options = Options()
        self.edge_options.add_argument("--start-maximized")  # 启动时最大化窗口
        self.edge_options.add_argument("--remote-allow-origins=*")  # 允许远程调试

        # 3. 启动 Edge 浏览器
        self.driver = webdriver.Edge(
            service=Service(self.driver_path),
            options=self.edge_options
        )
        self.driver.implicitly_wait(2)
    def getScreeShot(self):
        #创建屏幕截图
        #图片文件名称
        dirname = datetime.datetime.now().strftime("%Y-%m-%d")
        #判断dirname是否存在，若不存在则创建文件夹
        if not os.path.exists("../images/"+dirname):
            os.mkdir("../images/"+dirname)
        filename =sys._getframe().f_back.f_code.co_name+datetime.datetime.now().strftime("%Y-%m-%d-%H%M%S")+".png"
        self.driver.save_screenshot("../images/"+dirname+"/"+filename)

# 创建全局 Edge 驱动实例
ForumDriver = Driver()